<template>
  <div>
    <vxe-table
      border
      ref="tableRef"
      height="200"
      :data="tableData">
      <vxe-column type="checkbox" width="60"></vxe-column>
      <vxe-column field="name" title="Name" width="200">
        <template #header>
          <i :class="collapsable1 ? 'vxe-icon-square-minus' : 'vxe-icon-square-plus'" @click="collapsable1Event"></i>
          <span>Name</span>
        </template>
      </vxe-column>
      <vxe-column field="role" title="Role" :visible="false" width="200"></vxe-column>
      <vxe-column field="sex" title="Sex" :visible="false" width="200"></vxe-column>
      <vxe-column field="age" title="Age" width="200">
        <template #header>
          <i :class="collapsable2 ? 'vxe-icon-square-minus' : 'vxe-icon-square-plus'" @click="collapsable2Event"></i>
          <span>Age</span>
        </template>
      </vxe-column>
      <vxe-column field="rate" title="Rate" :visible="false" width="200"></vxe-column>
      <vxe-column field="address" title="Address" :visible="false" width="200"></vxe-column>
    </vxe-table>
  </div>
</template>

<script lang="ts" setup>
import { ref } from 'vue'
import { VxeTableInstance } from 'vxe-table'

interface RowVO {
  id: number
  name: string
  role: string
  sex: string
  age: number
  address: string
}

const tableRef = ref<VxeTableInstance<RowVO>>()

const collapsable1 = ref(false)
const collapsable2 = ref(false)
const tableData = ref<RowVO[]>([
  { id: 10001, name: 'Test1', role: 'Develop', sex: 'Man', age: 28, address: 'test abc' },
  { id: 10002, name: 'Test2', role: 'Test', sex: 'Women', age: 22, address: 'Guangzhou' },
  { id: 10003, name: 'Test3', role: 'PM', sex: 'Man', age: 32, address: 'Shanghai' },
  { id: 10004, name: 'Test4', role: 'Designer', sex: 'Women', age: 23, address: 'test abc' },
  { id: 10005, name: 'Test5', role: 'Develop', sex: 'Women', age: 30, address: 'Shanghai' },
  { id: 10006, name: 'Test6', role: 'Designer', sex: 'Women', age: 21, address: 'test abc' },
  { id: 10007, name: 'Test7', role: 'Test', sex: 'Man', age: 29, address: 'test abc' },
  { id: 10008, name: 'Test8', role: 'Develop', sex: 'Man', age: 35, address: 'test abc' }
])

const collapsable1Event = () => {
  const $table = tableRef.value
  if ($table) {
    const fields = ['role', 'sex']
    collapsable1.value = !collapsable1.value
    fields.forEach(field => {
      const column = $table.getColumnByField(field)
      if (column) {
        // 如果是直接修改实例属性，则需要调用 refreshColumn 刷新列
        column.visible = collapsable1.value
      }
    })
    $table.refreshColumn()
  }
}

const collapsable2Event = () => {
  const $table = tableRef.value
  if ($table) {
    const fields = ['rate', 'address']
    collapsable2.value = !collapsable2.value
    fields.forEach(field => {
      const column = $table.getColumnByField(field)
      if (column) {
        // 如果是直接修改实例属性，则需要调用 refreshColumn 刷新列
        column.visible = collapsable2.value
      }
    })
    $table.refreshColumn()
  }
}
</script>
